package com.rongmai.web.bo;

import com.rongmai.db.dto.CompanyFeatureDTO;
import com.rongmai.db.enums.UserRoleEnum;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
@Service
public class CompanyFutureBo {

    //查询投资、融资角色特性
    public static Map<String, List<CompanyFeatureDTO>> getTRFeature() {
        Map<String, List<CompanyFeatureDTO>> result = new HashMap<>();
        List<CompanyFeatureDTO> featureA = new ArrayList<>();
        CompanyFeatureDTO featureA1 = new CompanyFeatureDTO();
        featureA1.setKey("invest_industry");
        featureA1.setDesc("投资行业");
        featureA.add(featureA1);
        CompanyFeatureDTO featureA2 = new CompanyFeatureDTO();
        featureA2.setKey("invest_model");
        featureA2.setDesc("投资模式");
        featureA.add(featureA2);
        CompanyFeatureDTO featureA3 = new CompanyFeatureDTO();
        featureA3.setKey("invest_product");
        featureA3.setDesc("金融产品");
        featureA.add(featureA3);

        List<CompanyFeatureDTO> featureB = new ArrayList<>();
        CompanyFeatureDTO featureB1 = new CompanyFeatureDTO();
        featureB1.setKey("finance_industry");
        featureB1.setDesc("融资行业");
        featureA.add(featureB1);
        CompanyFeatureDTO featureB2 = new CompanyFeatureDTO();
        featureB2.setKey("finance_model");
        featureB2.setDesc("融资模式");
        featureA.add(featureB2);
        CompanyFeatureDTO featureB3 = new CompanyFeatureDTO();
        featureB3.setKey("finance_product");
        featureB3.setDesc("金融产品");
        featureB.add(featureB3);

        result.put(UserRoleEnum.INVEST.name(), featureA);
        result.put(UserRoleEnum.FINANCE.name(), featureB);
        return result;
    }
    //查询股权投资、融资角色特性
    public static Map<String, List<CompanyFeatureDTO>> getGQTRFeature() {
        Map<String, List<CompanyFeatureDTO>> result = new HashMap<>();
        List<CompanyFeatureDTO> featureA = new ArrayList<>();
        CompanyFeatureDTO featureA1 = new CompanyFeatureDTO();
        featureA1.setKey("invest_type");
        featureA1.setDesc("投资属性");
        featureA.add(featureA1);
        CompanyFeatureDTO featureA2 = new CompanyFeatureDTO();
        featureA2.setKey("invest_area");
        featureA2.setDesc("投资领域");
        featureA.add(featureA2);
        CompanyFeatureDTO featureA3 = new CompanyFeatureDTO();
        featureA3.setKey("invest_stage");
        featureA3.setDesc("投资阶段");
        featureA.add(featureA3);

        List<CompanyFeatureDTO> featureB = new ArrayList<>();
        CompanyFeatureDTO featureB1 = new CompanyFeatureDTO();
        featureB1.setKey("finance_type");
        featureB1.setDesc("融资属性");
        featureB.add(featureB1);
        CompanyFeatureDTO featureB2 = new CompanyFeatureDTO();
        featureB2.setKey("finance_area");
        featureB2.setDesc("融资领域");
        featureB.add(featureB2);
        CompanyFeatureDTO featureB3 = new CompanyFeatureDTO();
        featureB3.setKey("finance_stage");
        featureB3.setDesc("融资阶段");
        featureB.add(featureB3);

        result.put(UserRoleEnum.SHAREHOLD_INVEST.name(), featureA);
        result.put(UserRoleEnum.SHAREHOLD_FINANCE.name(), featureB);
        return result;
    }
}
